diff --git a/tests/mocks/database/db/driver.php b/tests/mocks/database/db/driver.php
index 9bf5231..cb18202 100644
--- a/tests/mocks/database/db/driver.php
+++ b/tests/mocks/database/db/driver.php
@@ -33,4 +33,4 @@
 	}
 }
 
-class CI_DB extends CI_DB_Driver {}
\ No newline at end of file
+class CI_DB extends Mock_Database_DB_QueryBuilder {}
\ No newline at end of file